animation in swift 4